Step Function Lambda Python